package com.qf.listener.simple;

import com.rabbitmq.client.Channel;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitHandler;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.concurrent.TimeUnit;

/**
 *
 * @RabbitListener: 指定要监听的队列名称
 * @author 千锋教育
 * @Company http://www.mobiletrain.org/
 * @Version 1.0
 */
@Component
@RabbitListener(queues = "simple_queue")
public class SimpleQueueListener {
    //StringRedisTemplate 相当于 RedisTemplate<String,String>
    //操作String 数据结构
    @Autowired
    StringRedisTemplate redisTemplate;

    @RabbitHandler
    public void consum(String msg, Channel channel, Message message){
        //获取消息id（唯一标识）
        String messageId = (String) message.getMessageProperties().getHeaders().get("spring_returned_message_correlation");
        //setIfAbsent:判断是否存在key， 如果存在key， 则说明都不做(返回false)
        // 如果key不存在，相当于set操作（返回true）
        // key : messageId  value: 0
        Boolean flag = redisTemplate.opsForValue().setIfAbsent(messageId, "0", 2, TimeUnit.MINUTES);
        //如果不存在，正常消费
        if(flag) {
            System.out.println("消费者消费："+msg);
            //消费完成后， 修改状态为 1
            redisTemplate.opsForValue().set(messageId, "1");
            //手动ack
            try {
                channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
            } catch (IOException e) {
                e.printStackTrace();
            }
            //删除redis中的 messageId信息
            redisTemplate.delete(messageId);
        }
        //返回false，说明存在此messageId， 说明已经被消费
        else {
            //获取messageId对应的value
            String value = redisTemplate.opsForValue().get(messageId);
            //如果value为1， 说明消费成功， 则直接手动ack（确认已消费）
            if("1".equalsIgnoreCase(value)) {
                try {
                    channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                //手动ack完成后，删除key
                redisTemplate.delete(messageId);
            }
            //如果value为0， 说明正在消费
        }
    }
}
